# docker build -t test-sts:v2 .

FROM nvcr.io/nvidia/pytorch:25.04-py3
WORKDIR /workspace
RUN pip install --no-cache-dir transformers accelerate -i https://pypi.tuna.tsinghua.edu.cn/simple
CMD ["/bin/sleep", "infinity"]

